package com.example.config.handle;

import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import com.example.dao.dataobject.BaseDO;
import org.apache.ibatis.reflection.MetaObject;

import java.time.LocalDateTime;
import java.util.Objects;

/**
 * @author Houger.
 * @date 2025年01月15日 15:18.
 */
public abstract class AbstractDBFieldHandler implements MetaObjectHandler {

    @Override
    public void insertFill(MetaObject metaObject) {
        if (Objects.nonNull(metaObject) && metaObject.getOriginalObject() instanceof BaseDO) {
            BaseDO baseDO = (BaseDO) metaObject.getOriginalObject();

            LocalDateTime current = LocalDateTime.now();
            // 创建时间为空，则以当前时间为插入时间
            if (Objects.isNull(baseDO.getCreateTime())) {
                baseDO.setCreateTime(current);
            }
            // 更新时间为空，则以当前时间为更新时间
            baseDO.setUpdateTime(current);

            String operator = getOperator();
            // 当前登录用户不为空，创建人为空，则当前登录用户为创建人
            if (Objects.nonNull(operator) && Objects.isNull(baseDO.getCreator())) {
                baseDO.setCreator(operator);
            }
            // 当前登录用户不为空，更新人为空，则当前登录用户为更新人
            if (Objects.nonNull(operator) && Objects.isNull(baseDO.getUpdater())) {
                baseDO.setUpdater(operator);
            }
        }
    }

    @Override
    public void updateFill(MetaObject metaObject) {
        if (Objects.nonNull(metaObject) && metaObject.getOriginalObject() instanceof BaseDO) {
            // 更新时间为空，则以当前时间为更新时间
            Object modifyTime = getFieldValByName("updateTime", metaObject);
            if (Objects.isNull(modifyTime)) {
                setFieldValByName("updateTime", LocalDateTime.now(), metaObject);
            }

            // 当前登录用户不为空，更新人为空，则当前登录用户为更新人
            Object modifier = getFieldValByName("updater", metaObject);
            String operator = getOperator();
            if (Objects.nonNull(operator) && Objects.isNull(modifier)) {
                setFieldValByName("updater", operator, metaObject);
            }
        }
    }

    protected abstract String getOperator();
}

